import torchvision
import  torch


vgg16 = torchvision.models.vgg16(pretrained=True)

# 1.网络模型 + 参数
torch.save(vgg16,"../../../dataSet/model/vgg16_method1.pdh")


# 2. 只存训练参数   官方推荐的
torch.save(vgg16.state_dict(),"../../../dataSet/model/vgg16_method2.pdh")


